package com.vv.study.factory;

import com.alibaba.druid.util.StringUtils;
import com.vv.study.anno.Autowired;
import com.vv.study.anno.Service;
import com.vv.study.anno.Transactional;
import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.Element;
import org.dom4j.io.SAXReader;
import org.reflections.Reflections;

import java.io.InputStream;
import java.lang.annotation.Annotation;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;

/**
 * @author 应癫
 *
 * 工厂类，生产对象（使用反射技术）
 */
public class BeanFactory {

    /**
     * 任务一：读取解析xml，通过反射技术实例化对象并且存储待用（map集合）
     * 任务二：对外提供获取实例对象的接口（根据id获取）
     */

    private static Map<String,Object> map = new HashMap<>();  // 存储对象


//    static {
//        // 任务一：读取解析xml，通过反射技术实例化对象并且存储待用（map集合）
//        // 加载xml
//        InputStream resourceAsStream = BeanFactory.class.getClassLoader().getResourceAsStream("beans.xml");
//        // 解析xml
//        SAXReader saxReader = new SAXReader();
//        try {
//            Document document = saxReader.read(resourceAsStream);
//            Element rootElement = document.getRootElement();
//            List<Element> beanList = rootElement.selectNodes("//bean");
//            for (int i = 0; i < beanList.size(); i++) {
//                Element element =  beanList.get(i);
//                // 处理每个bean元素，获取到该元素的id 和 class 属性
//                String id = element.attributeValue("id");        // accountDao
//                String clazz = element.attributeValue("class");  // com.vv.study.dao.impl.JdbcAccountDaoImpl
//                // 通过反射技术实例化对象
//                Class<?> aClass = Class.forName(clazz);
//                Object o = aClass.newInstance();  // 实例化之后的对象
//
//                // 存储到map中待用
//                map.put(id,o);
//
//            }
//
//            // 实例化完成之后维护对象的依赖关系，检查哪些对象需要传值进入，根据它的配置，我们传入相应的值
//            // 有property子元素的bean就有传值需求
//            List<Element> propertyList = rootElement.selectNodes("//property");
//            // 解析property，获取父元素
//            for (int i = 0; i < propertyList.size(); i++) {
//                Element element =  propertyList.get(i);   //<property name="AccountDao" ref="accountDao"></property>
//                String name = element.attributeValue("name");
//                String ref = element.attributeValue("ref");
//
//                // 找到当前需要被处理依赖关系的bean
//                Element parent = element.getParent();
//
//                // 调用父元素对象的反射功能
//                String parentId = parent.attributeValue("id");
//                Object parentObject = map.get(parentId);
//                // 遍历父对象中的所有方法，找到"set" + name
//                Method[] methods = parentObject.getClass().getMethods();
//                for (int j = 0; j < methods.length; j++) {
//                    Method method = methods[j];
//                    if(method.getName().equalsIgnoreCase("set" + name)) {  // 该方法就是 setAccountDao(AccountDao accountDao)
//                        method.invoke(parentObject,map.get(ref));
//                    }
//                }
//
//                // 把处理之后的parentObject重新放到map中
//                map.put(parentId,parentObject);
//
//            }
//
//
//        } catch (DocumentException e) {
//            e.printStackTrace();
//        } catch (ClassNotFoundException e) {
//            e.printStackTrace();
//        } catch (IllegalAccessException e) {
//            e.printStackTrace();
//        } catch (InstantiationException e) {
//            e.printStackTrace();
//        } catch (InvocationTargetException e) {
//            e.printStackTrace();
//        }
//
//    }

    static {
        // 任务一：扫描工程下所有含自定义注解类，通过反射技术实例化对象并且存储待用（map集合）
        try {
            // 扫描包
            Reflections reflections = new Reflections("com.vv.study");

            //获取包下所有Service注解类
            Set<Class<?>> serviceAnnoClazzs = reflections.getTypesAnnotatedWith(Service.class);

            for (Class clazz : serviceAnnoClazzs) {
                //获取注解
                Service annotation = (Service)clazz.getAnnotation(Service.class);

                // 设置bean id
                String value = annotation.value();
                // 如果没有自定义bean id，则设置bean id为class name
                if ("".equals(value)) {
                    //由于getName获取的是全限定类名，所以要分割去掉前面包名部分
                    String[] names = clazz.getName().split("\\.");
                    value = names[names.length-1];
                }

                // 通过反射技术实例化对象
                Object o = clazz.newInstance();  // 实例化之后的对象

                // 存储到map中待用
                map.put(value,o);

            }

            // 实例化完成之后维护对象的依赖关系，检查哪些对象通过需要Autowired传值注入
            for(String key : map.keySet()){
                //@Service注解标识的类
                Object service = map.get(key);

                //获取对应的class
                Class clazz = service.getClass();

                //获取所有的属性值
                Field[] fields = clazz.getDeclaredFields();

                //获取所有的方法
                Method[] methods = clazz.getMethods();

                //遍历所有的属性，判断是否有Autowired注解，如果有则通过set方法注入
                for(Field field : fields){
                    if(field.isAnnotationPresent(Autowired.class)){
                        String[] names = field.getType().getName().split("\\.");
                        String name = names[names.length-1];
                        // 遍历父对象中的所有方法，找到"set" + name
                        for (int j = 0; j < methods.length; j++) {
                            Method method = methods[j];
                            if(method.getName().equalsIgnoreCase("set" + name)) {  // 该方法就是 setAccountDao(AccountDao accountDao)
                                method.invoke(service,map.get(name));
                                break;
                            }
                        }
                    }
                }

                // 把处理之后的service重新放到map中
                map.put(key,service);

            }


            // 最后看是否需要有Transactional注解，如果有则实现事务控制
            for(String key : map.keySet()){

                Object service = map.get(key);
                //获取对应的class
                Class clazz = service.getClass();
                if(clazz.isAnnotationPresent(Transactional.class)){
                    //获取代理工厂
                    ProxyFactory proxyFactory = (ProxyFactory) BeanFactory.getBean("ProxyFactory");
                    //判断对象是否实现接口
                    //获取类c实现的所有接口
                    Class[] face = clazz.getInterfaces();
                    if(face!=null&&face.length>0){
                        //实现使用JDK
                        service = proxyFactory.getJdkProxy(service);
                    }else{
                        //没实现使用CGLIB
                        service = proxyFactory.getCglibProxy(service);
                    }
                }
                // 把处理之后的object重新放到map中
                map.put(key,service);
            }

        } catch (IllegalAccessException e) {
            e.printStackTrace();
        } catch (InstantiationException e) {
            e.printStackTrace();
        } catch (InvocationTargetException e) {
            e.printStackTrace();
        }

    }


    // 任务二：对外提供获取实例对象的接口（根据id获取）
    public static  Object getBean(String id) {
        return map.get(id);
    }

}
